package com.ruoyu.cloud.ai.mcp.config;

import com.ruoyu.cloud.ai.mcp.service.BookService;
import org.springframework.ai.tool.ToolCallbackProvider;
import org.springframework.ai.tool.method.MethodToolCallbackProvider;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

@Configuration
public class McpServerConfig {

    /**
     * 注册工具回调提供者，将BookService中的@Tool方法暴露为MCP工具
     *
     * @param bookService 图书服务
     * @return 工具回调提供者
     */
    @Bean
    public ToolCallbackProvider bookToolCallbackProvider(BookService bookService) {
        return MethodToolCallbackProvider.builder()
                .toolObjects(bookService)
                .build();
    }
}
